// Copyright (C) 2018-2021 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//

#pragma once

#include <cmath>
#include <numeric>
#include <vector>

#include "ngraph/shape.hpp"

namespace ngraph
{
    namespace runtime
    {
        namespace reference
        {
            void shuffle_channels(const char* arg,
                                  char* out,
                                  const Shape& data_shape,
                                  size_t elem_size,
                                  const int64_t axis,
                                  const int64_t group);
        } // namespace reference
    }     // namespace runtime
} // namespace ngraph
